"""
模型模块，包含所有与图像着色和分类相关的模型定义
"""

from models.colorization_model import ColorizationGenerator, ColorizationDiscriminator, UNetDown, UNetUp
from models.image_classifier import ImageClassifier
from models.model_manager import ModelManager

__all__ = [
    'ColorizationGenerator',
    'ColorizationDiscriminator',
    'UNetDown', 
    'UNetUp',
    'ImageClassifier',
    'ModelManager',
] 